@echo off
chcp 65001 >nul
echo ========================================
echo    游戏按键录制与回放系统
echo ========================================
echo.

set PYTHON_EXE=.venv\Scripts\python.exe

if not exist %PYTHON_EXE% (
    echo ❌ 未找到Python虚拟环境
    echo 请先运行: python -m venv .venv
    echo 然后: .venv\Scripts\pip install -r requirements.txt
    pause
    exit /b 1
)

:MENU
echo 请选择操作:
echo [1] 列出可用窗口
echo [2] 开始录制 (需要窗口标题)
echo [3] 分析录制文件
echo [4] 回放录制文件
echo [5] 退出
echo.
set /p choice=请输入选项 (1-5): 

if "%choice%"=="1" goto LIST_WINDOWS
if "%choice%"=="2" goto START_RECORDING
if "%choice%"=="3" goto ANALYZE_FILE
if "%choice%"=="4" goto REPLAY_FILE
if "%choice%"=="5" goto EXIT
echo 无效选项，请重新选择
goto MENU

:LIST_WINDOWS
echo.
echo 📋 列出所有可用窗口...
%PYTHON_EXE% auto_screenshot.py --list
echo.
pause
goto MENU

:START_RECORDING
echo.
set /p window_title=请输入窗口标题关键词: 
if "%window_title%"=="" (
    echo 窗口标题不能为空
    goto START_RECORDING
)

set /p interval=请输入截图间隔秒数 (默认2.0): 
if "%interval%"=="" set interval=2.0

echo.
echo 🎬 开始录制...
echo 窗口: %window_title%
echo 截图间隔: %interval%秒
echo 按 Ctrl+C 停止录制
echo.
%PYTHON_EXE% auto_screenshot.py --title "%window_title%" --record-keys --interval %interval%
echo.
pause
goto MENU

:ANALYZE_FILE
echo.
set /p json_file=请输入JSON文件路径: 
if "%json_file%"=="" (
    echo 文件路径不能为空
    goto ANALYZE_FILE
)

if not exist "%json_file%" (
    echo 文件不存在: %json_file%
    goto ANALYZE_FILE
)

echo.
echo 📊 分析录制文件...
%PYTHON_EXE% replay_events.py --analyze "%json_file%"
echo.
pause
goto MENU

:REPLAY_FILE
echo.
set /p json_file=请输入JSON文件路径: 
if "%json_file%"=="" (
    echo 文件路径不能为空
    goto REPLAY_FILE
)

if not exist "%json_file%" (
    echo 文件不存在: %json_file%
    goto REPLAY_FILE
)

echo.
echo 请选择回放模式:
echo [1] 预览模式 (不执行)
echo [2] 实际回放 (原速)
echo [3] 实际回放 (自定义速度)
set /p replay_mode=请选择 (1-3): 

if "%replay_mode%"=="1" (
    echo.
    echo 🔍 预览模式...
    %PYTHON_EXE% replay_events.py --preview "%json_file%"
) else if "%replay_mode%"=="2" (
    echo.
    echo 🎬 开始原速回放...
    %PYTHON_EXE% replay_events.py "%json_file%"
) else if "%replay_mode%"=="3" (
    set /p speed=请输入速度倍率 (如 0.5, 1.0, 2.0): 
    if "!speed!"=="" set speed=1.0
    echo.
    echo 🎬 开始 !speed!x 速度回放...
    %PYTHON_EXE% replay_events.py "%json_file%" --speed !speed!
) else (
    echo 无效选项
    goto REPLAY_FILE
)

echo.
pause
goto MENU

:EXIT
echo.
echo 👋 谢谢使用！
exit /b 0
